
import os
import glob
import shutil

dir1 = '\\D:\\Desktop\\工程实践\\瑕疵检测数据集\\初赛1\\xuelang_round1_answer_a_20180808\\'
dir2 = '\\D:\\Desktop\\工程实践\\瑕疵检测数据集\\初赛1\\xuelang_round1_answer_b_20180808\\'
dir3 = '\\D:\\Desktop\\工程实践\\瑕疵检测数据集\\初赛1\\xuelang_round1_test_a_20180709\\'
dir4 = '\\D:\\Desktop\\工程实践\\瑕疵检测数据集\\初赛1\\xuelang_round1_test_b\\'

# a文件夹处理
# 移动瑕疵图片到指定分类文件夹
xml_list = glob.glob(dir1 + "*\\*.xml")
for xml_path in xml_list:
    jpg_name = xml_path.split('\\')[-1].replace('xml', 'jpg')
    cat_name = xml_path.split('\\')[-2] + '\\'
    shutil.move(dir3 + jpg_name, dir1 + cat_name + jpg_name)

# a里面竟然没有正常文件夹，代码建一个
os.makedirs(dir1+'正常')

jpg_list = glob.glob(dir3 + "*.jpg")
for jpg_path in jpg_list:
    jpg_name = jpg_path.split('\\')[-1]
    shutil.move(jpg_path, dir1 + '正常\\' + jpg_name)

# b文件夹处理
# 移动瑕疵图片到指定分类文件夹
xml_list = glob.glob(dir2 + "*\\*.xml")
for xml_path in xml_list:
    jpg_name = xml_path.split('\\')[-1].replace('xml', 'jpg')
    cat_name = xml_path.split('\\')[-2] + '\\'
    shutil.move(dir4 + jpg_name, dir2 + cat_name + jpg_name)
# 移动正常图片
jpg_list = glob.glob(dir4 + "*.jpg")
for jpg_path in jpg_list:
    jpg_name = jpg_path.split('\\')[-1]
    shutil.move(jpg_path, dir2 + '正常\\' + jpg_name)

# 删除剩下的两个空文件夹
os.removedirs(dir3)
os.removedirs(dir4)
